package club.xiaojiawei.dp;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 6/6/22 12:15 AM
 * @question 123. 买卖股票的最佳时机 III
 * @description 给定一个数组，它的第 i 个元素是一支给定的股票在第 i 天的价格。
 * 设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。
 * 注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
 */
public class MaxProfit123 {

    public static void main(String[] args) {
        MaxProfit123 test = new MaxProfit123();
        int result = test.maxProfit(new int[]{1, 8, 3, 5, 4, 8});
        System.out.println("最大利润："+result);
    }

    /**
     * 官方-dp
     * 时间复杂度 O(n)
     * 空间复杂度 O(1)
     * @param prices
     * @return
     */
    public int maxProfit(int[] prices) {
        int n = prices.length;
        int buy1 = -prices[0], sell1 = 0;
        int buy2 = -prices[0], sell2 = 0;
        for (int i = 1; i < n; ++i) {
//            只进行利润最大的一次交易
            buy1 = Math.max(buy1, -prices[i]);
            sell1 = Math.max(sell1, buy1 + prices[i]);
//            第一次交易赚的钱作为本钱，只进行利润第二大的一次交易（为什么不是第一大，因为第一大的代码在此之前，已经交易过了）
            buy2 = Math.max(buy2, sell1 - prices[i]);
            sell2 = Math.max(sell2, buy2 + prices[i]);
        }
        return sell1;
    }


}
